package log

import (
	"fmt"
	"io"
	"log"
	"os"
	"path"
	"time"

	"github.com/sirupsen/logrus"
)

var LogrusObj *logrus.Logger

func Init() {
	if LogrusObj != nil {
		src, _ := setOutputFile()
		// 设置输出
		LogrusObj.Out = src
		return
	}
	// 实例化
	logger := logrus.New()
	src, _ := setOutputFile()
	// 设置输出
	logger.Out = src
	// 设置日志级别
	logger.SetLevel(logrus.DebugLevel)
	// 设置日志格式
	logger.SetFormatter(&logrus.JSONFormatter{
		TimestampFormat: "2006-01-02 15:04:05",
	})

	LogrusObj = logger
}

func setOutputFile() (*os.File, error) {
	now := time.Now()
	logFilePath := ""
	if dir, err := os.Getwd(); err == nil {
		logFilePath = dir + "/logs/"
	}
	_, err := os.Stat(logFilePath)
	if os.IsNotExist(err) {
		if err := os.MkdirAll(logFilePath, 0777); err != nil {
			log.Println(err.Error())
			return nil, err
		}
	}
	logFileName := now.Format("2006-01-02") + ".log"
	// 日志文件
	fileName := path.Join(logFilePath, logFileName)
	if _, err := os.Stat(fileName); err != nil {
		if _, err := os.Create(fileName); err != nil {
			log.Println(err.Error())
			return nil, err
		}
	}
	// 写入文件
	src, err := os.OpenFile(fileName, os.O_APPEND|os.O_WRONLY, os.ModeAppend)
	if err != nil {
		log.Println(err)
		return nil, err
	}
	return src, nil
}

// 返回基于当前时间的日志文件名
func GetLogFilename() string {
	a := fmt.Sprintf("lion-app_%s.log", time.Now().Format("2006-01-02_15"))
	return a
}

// 重新创建日志文件
func RotateLog() {
	newFilename := GetLogFilename()

	// 打开源文件
	sourceFile, err := os.Open("lion-app.log")
	if err != nil {
		fmt.Printf("Error opening source file: %v\n", err)
		return
	}
	defer sourceFile.Close()

	// 创建目标文件
	destinationFile, err := os.Create(newFilename)
	if err != nil {
		fmt.Printf("Error creating destination file: %v\n", err)
		return
	}
	defer destinationFile.Close()

	// 复制文件内容
	bytesCopied, err := io.Copy(destinationFile, sourceFile)
	if err != nil {
		fmt.Printf("Error copying file: %v\n", err)
		return
	}
	fmt.Printf("Copied %d bytes from lion-app.log to %s\n", bytesCopied, newFilename)

	// 截断文件为 0 字节，清空内容
	err1 := os.Truncate("lion-app.log", 0)
	if err1 != nil {
		fmt.Printf("Error truncating file: %v\n", err1)
		return
	}

	fmt.Println("app.log content has been cleared.")
}
